import tensorflow as tf
import torch
gpu_device_name = tf.test.gpu_device_name()
available = tf.test.is_gpu_available()
print('gpu count:',torch.cuda.device_count())
print('deviceName:',gpu_device_name)
print('gpu is available',available)